from utils.sftp_clean_data import SftpCleanData

if __name__ == '__main__':
    test = SftpCleanData()
    basedir =  r"I:\project\sun-procurement\sunshin-procurement-python\flaskr\testdemo\天方消费上传数据"

    # 插入数据库示例
    folders_to_process = [
        rf"{basedir}\huanggangsftp",
        rf"{basedir}\huangshisftp",
        rf"{basedir}\xiantaosftp",
        rf"{basedir}\shennongjiasftp",
        rf"{basedir}\shiyansftp",
        rf"{basedir}\xianningsftp",
        rf"{basedir}\enshisftp",
        rf"{basedir}\yichangsftp",
        rf"{basedir}\qianjiangsftp",
    ]


    for folder in folders_to_process:
        test.batch_insert_from_folder(folder+'_cleanfile')